package org.yzhzc.friend.service.user.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONUtil;
import com.alibaba.fastjson2.JSON;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.yzhzc.api.domain.UserExeResult;
import org.yzhzc.api.domain.dto.JudgeSubmitDTO;
import org.yzhzc.api.domain.vo.UserQuestionResultVO;
import org.yzhzc.api.service.RemoteJudgeService;
import org.yzhzc.common.core.constants.Constants;
import org.yzhzc.common.core.domain.R;
import org.yzhzc.common.core.enums.ProgramType;
import org.yzhzc.common.core.enums.QuestionResType;
import org.yzhzc.common.core.enums.ResultCode;
import org.yzhzc.common.core.utils.ThreadLocalUtil;
import org.yzhzc.common.security.exception.ServiceException;
import org.yzhzc.friend.domain.question.QuestionCase;
import org.yzhzc.friend.domain.question.entity.Question;
import org.yzhzc.friend.domain.question.es.QuestionES;
import org.yzhzc.friend.domain.user.dto.UserSubmitDTO;
import org.yzhzc.friend.domain.user.entity.UserSubmit;
import org.yzhzc.friend.mapper.question.QuestionMapper;
import org.yzhzc.friend.mapper.user.UserSubmitMapper;
import org.yzhzc.friend.rabbit.JudgeProducer;
import org.yzhzc.friend.repository.question.QuestionRepository;
import org.yzhzc.friend.service.user.IUserQuestionService;

import java.util.List;

/**
 * 用户题目服务实现类。
 * 提供用户提交代码、查询判题结果等功能。
 */
@Slf4j
@Service
public class UserQuestionServiceImpl implements IUserQuestionService {

    @Autowired // 题目 ES 仓库
    private QuestionRepository questionRepository;

    @Autowired // 题目数据访问接口
    private QuestionMapper questionMapper;

    @Autowired // 用户提交记录数据访问接口
    private UserSubmitMapper userSubmitMapper;

    @Autowired // 远程判题服务
    private RemoteJudgeService remoteJudgeService;

    @Autowired // 判题任务生产者
    private JudgeProducer judgeProducer;

    /**
     * 用户提交代码并进行判题。
     *
     * @param submitDTO 用户提交信息，包含代码、题目 ID、编程语言类型等
     * @return 判题结果，封装在 R 对象中
     * @throws ServiceException 如果编程语言类型不支持，则抛出异常
     */
    @Override
    public R<UserQuestionResultVO> submit(UserSubmitDTO submitDTO) {
        Integer programType = submitDTO.getProgramType();
        if (ProgramType.JAVA.getValue().equals(programType)) { // 判断是否为 Java 语言
            JudgeSubmitDTO judgeSubmitDTO = assembleJudgeSubmitDTO(submitDTO); // 组装判题提交信息
            return remoteJudgeService.doJudgeJavaCode(judgeSubmitDTO); // 调用远程判题服务
        }
        throw new ServiceException(ResultCode.FAILED_NOT_SUPPORT_PROGRAM); // 抛出不支持编程语言的异常
    }

    /**
     * 用户提交代码并通过 RabbitMQ 发送判题任务。
     *
     * @param submitDTO 用户提交信息，包含代码、题目 ID、编程语言类型等
     * @return 是否成功发送判题任务
     * @throws ServiceException 如果编程语言类型不支持，则抛出异常
     */
    @Override
    public boolean rabbitSubmit(UserSubmitDTO submitDTO) {
        Integer programType = submitDTO.getProgramType();
        if (ProgramType.JAVA.getValue().equals(programType)) { // 判断是否为 Java 语言
            JudgeSubmitDTO judgeSubmitDTO = assembleJudgeSubmitDTO(submitDTO); // 组装判题提交信息
            judgeProducer.produceMsg(judgeSubmitDTO); // 发送判题任务到 RabbitMQ
            return true;
        }
        throw new ServiceException(ResultCode.FAILED_NOT_SUPPORT_PROGRAM); // 抛出不支持编程语言的异常
    }

    /**
     * 查询用户判题结果。
     *
     * @param examId     考试 ID
     * @param questionId 题目 ID
     * @param currentTime 当前时间
     * @return 用户判题结果视图对象
     */
    @Override
    public UserQuestionResultVO exeResult(Long examId, Long questionId, String currentTime) {
        Long userId = ThreadLocalUtil.get(Constants.USER_ID, Long.class); // 获取当前用户 ID
        UserSubmit userSubmit = userSubmitMapper.selectCurrentUserSubmit(userId, examId, questionId, currentTime); // 查询用户提交记录
        UserQuestionResultVO resultVO = new UserQuestionResultVO(); // 创建判题结果视图对象
        if (userSubmit == null) { // 如果用户未提交
            resultVO.setPass(QuestionResType.IN_JUDGE.getValue()); // 设置为判题中
        } else {
            resultVO.setPass(userSubmit.getPass()); // 设置判题结果
            resultVO.setExeMessage(userSubmit.getExeMessage()); // 设置执行信息
            if (StrUtil.isNotEmpty(userSubmit.getCaseJudgeRes())) { // 如果用例判题结果不为空
                resultVO.setUserExeResultList(JSON.parseArray(userSubmit.getCaseJudgeRes(), UserExeResult.class)); // 解析用例判题结果
            }
        }
        return resultVO;
    }

    /**
     * 组装判题提交信息。
     *
     * @param submitDTO 用户提交信息
     * @return 判题提交信息
     */
    private JudgeSubmitDTO assembleJudgeSubmitDTO(UserSubmitDTO submitDTO) {
        Long questionId = submitDTO.getQuestionId(); // 获取题目 ID
        JudgeSubmitDTO judgeSubmitDTO = new JudgeSubmitDTO(); // 创建判题提交信息对象

        // 从 ES 中查询题目信息
        QuestionES questionES = questionRepository.findById(questionId).orElse(null);
        if (questionES != null) { // 如果 ES 中存在题目信息
            BeanUtil.copyProperties(questionES, judgeSubmitDTO); // 复制属性
        } else { // 如果 ES 中不存在题目信息
            Question question = questionMapper.selectById(questionId); // 从数据库中查询题目信息
            BeanUtil.copyProperties(question, judgeSubmitDTO); // 复制属性
            questionES = new QuestionES(); // 创建 ES 实体对象
            BeanUtil.copyProperties(question, questionES); // 复制属性
            questionRepository.save(questionES); // 保存到 ES
        }

        judgeSubmitDTO.setUserId(ThreadLocalUtil.get(Constants.USER_ID, Long.class)); // 设置用户 ID
        judgeSubmitDTO.setExamId(submitDTO.getExamId());           // 设置考试 ID
        judgeSubmitDTO.setProgramType(submitDTO.getProgramType()); // 设置编程语言类型
        judgeSubmitDTO.setUserCode(codeConnect(submitDTO.getUserCode(), questionES.getMainFuc())); // 拼接用户代码和主函数

        // 获取输入列表
        List<QuestionCase> questionCaseList = JSONUtil.toList(questionES.getQuestionCase(), QuestionCase.class); // 解析题目用例
        List<String> inputList = questionCaseList.stream().map(QuestionCase::getInput).toList();
        judgeSubmitDTO.setInputList(inputList); // 设置输入列表

        // 获取输出列表
        List<String> outputList = questionCaseList.stream().map(QuestionCase::getOutput).toList();
        judgeSubmitDTO.setOutputList(outputList); // 设置输出列表

        return judgeSubmitDTO;
    }

    /**
     * 拼接用户代码和主函数。
     *
     * @param userCode 用户代码
     * @param mainFunc 主函数
     * @return 拼接后的代码
     * @throws ServiceException 如果拼接失败，则抛出异常
     */
    private String codeConnect(String userCode, String mainFunc) {
        String targetCharacter = "}"; // 目标字符
        int targetLastIndex = userCode.lastIndexOf(targetCharacter); // 查找目标字符的最后位置
        if (targetLastIndex != -1) { // 如果找到目标字符
            return userCode.substring(0, targetLastIndex) + "\n" + mainFunc + "\n" + userCode.substring(targetLastIndex); // 拼接代码
        }
        throw new ServiceException(ResultCode.FAILED); // 抛出异常
    }
}